
#include <stdio.h>
#include <unistd.h>
#include <string.h>
#include <strings.h>
#include <stdlib.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <net/if.h>
#include <sys/ioctl.h>
#include <errno.h>
#include <fcntl.h>
#include <sys/select.h>

// 客户端，主动发起链接请求

int main(int argc, char const *argv[]) // ./client IP PORT
{
    if(argc != 3)
    {
        printf("请指定服务器IP和端口\n");
        exit(0);
    }

    // 1，创建TCP
    int sockfd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

    // 2，准备好服务器的地址
    struct sockaddr_in addr;
    socklen_t len = sizeof(addr);
    bzero(&addr, len);
    addr.sin_family = AF_INET;
    addr.sin_addr.s_addr = inet_addr(argv[1]);
    addr.sin_port = htons(atoi(argv[2]));

    // 3，发起连接请求
    connect(sockfd, (struct sockaddr *)&addr, len);

    // 4，采用多路复用的方式，同时监听标准输入、sockfd上的就绪状态
    char wbuf[100];
    char rbuf[100];
    while(1)
    {
        fd_set rset, wset, eset;
        FD_ZERO(&rset);
        FD_ZERO(&wset);
        FD_ZERO(&eset);
        FD_SET(STDIN_FILENO, &rset);
        FD_SET(sockfd, &rset);

        // 同时监控键盘和套接字的读就绪
        select(sockfd+1, &rset, &wset, &eset, NULL);

        // 1，键盘有输入了，马上要将输入信息发给对方
        if(FD_ISSET(STDIN_FILENO, &rset))
        {
            fgets(wbuf, 100, stdin);
            send(sockfd, wbuf, strlen(wbuf),0);
        }

        // 2，对方有数据来了，马上要读出来并输出出来
        if(FD_ISSET(sockfd, &rset))
        {
            bzero(rbuf, 100);
            if(recv(sockfd, rbuf, 100,0) > 0)
                printf("收到：%s", rbuf);
            else
                exit(0);
        }
    }

    // 5，关闭socket
    close(sockfd);

    return 0;
}
